import base64

from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad

# AES加密函数，使用Base64编码
def aes_encrypt_b64(encrypted_content, encryption_key):
    cipher = AES.new(encryption_key, AES.MODE_ECB)  # 使用ECB模式
    padded_data = pad(encrypted_content.encode(), AES.block_size)  # 数据填充
    encrypted_data = cipher.encrypt(padded_data)  # 加密
    return base64.b64encode(encrypted_data).decode()  # 使用Base64编码

# AES解密函数，使用Base64解码
def aes_decrypt_b64(cipher_text_b64, encryption_key):
    cipher_text = base64.b64decode(cipher_text_b64)  # 使用Base64解码
    cipher = AES.new(encryption_key, AES.MODE_ECB)  # 使用ECB模式
    decrypted_padded_data = cipher.decrypt(cipher_text)  # 解密
    return unpad(decrypted_padded_data, AES.block_size).decode()  # 去除填充并返回原始字符串